package com.map.core.utils;

import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.LinkedList;
import java.util.List;

/**
 * <p>Title: ImportExcelUtil</p>
 * <p>Description: </p>
 * @author DaiSn
 * @date 2019年3月13日
 */
public class ImportExcelUtil {
	public static List<List<Object>> importExcel(File file) throws IOException {
		String fileName = file.getName();
		String extension = fileName.lastIndexOf(".") == -1 ? "" : fileName.substring(fileName.lastIndexOf(".") + 1);
		if ("xls".equals(extension)) {
			return readXLSExcel(file);
		} else if ("xlsx".equals(extension)) {
			return readXLSXExcel(file);
		} else {
			throw new IOException("不支持的文件类型");
		}
	}

	/**
	 * 读取 office 2003 excel
	 *
	 * @throws IOException
	 * @throws FileNotFoundException
	 */
	private static List<List<Object>> readXLSExcel(File file) throws IOException {
		List<List<Object>> list = new LinkedList<List<Object>>();
		FileInputStream fis = new FileInputStream(file);
		HSSFWorkbook hwb = new HSSFWorkbook(fis);
		HSSFSheet sheet = hwb.getSheetAt(0);
		Object value = null;
		HSSFRow row = null;
		HSSFCell cell = null;
		try{
			for (int i = sheet.getFirstRowNum(); i <= sheet.getPhysicalNumberOfRows(); i++) {
				row = sheet.getRow(i);
				if (row == null) {
					continue;
				}
				List<Object> linked = new LinkedList<Object>();
				for (int j = row.getFirstCellNum(); j <= row.getLastCellNum(); j++) {
					cell = row.getCell(j);
					if (cell == null ||cell.getCellType()== CellType.BLANK) {
						linked.add("");
						continue;
					}
					DecimalFormat df = new DecimalFormat("0");// 格式化 number String
					// 字符
					SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");// 格式化日期字符串
					DecimalFormat nf = new DecimalFormat("0");// 格式化数字
					switch (cell.getCellType()) {
						case STRING:
							// System.out.println(i+"行"+j+" 列 is String type");
							value = cell.getStringCellValue();
							break;
						case NUMERIC:
							// System.out.println(i+"行"+j+" 列 is Number type ;
							// DateFormt:"+cell.getCellStyle().getDataFormatString());
							if ("@".equals(cell.getCellStyle().getDataFormatString())) {
								value = df.format(cell.getNumericCellValue());
							} else if ("General".equals(cell.getCellStyle().getDataFormatString())) {
								value = nf.format(cell.getNumericCellValue());
							} else {
								value = sdf.format(HSSFDateUtil.getJavaDate(cell.getNumericCellValue()));
							}
							break;
						case BOOLEAN:
							// System.out.println(i+"行"+j+" 列 is Boolean type");
							value = cell.getBooleanCellValue();
							break;
						case BLANK:
							// System.out.println(i+"行"+j+" 列 is Blank type");
							value = "";
							break;
						default:
							// System.out.println(i+"行"+j+" 列 is default type");
							value = cell.toString();
					}
					if (value == null || "".equals(value)) {
						continue;
					}
					linked.add(value);
				}
				list.add(linked);
			}
		}finally {
			if(hwb!=null){
				hwb.close();
			}
			if(fis!=null){
				fis.close();
			}
		}
		return list;
	}

	/**
	 * 读取Office 2007 excel
	 */
	private static List<List<Object>> readXLSXExcel(File file) throws IOException {
		List<List<Object>> list = new LinkedList<List<Object>>();
		FileInputStream fis = new FileInputStream(file);
		// 构造 XSSFWorkbook 对象，strPath 传入文件路径
		XSSFWorkbook xwb = new XSSFWorkbook(fis);
		// 读取第一章表格内容
		XSSFSheet sheet = xwb.getSheetAt(0);
		Object value = null;
		XSSFRow row = null;
		XSSFCell cell = null;
		boolean flag = false;
		try{
			for (int i = sheet.getFirstRowNum(); i <= sheet.getPhysicalNumberOfRows(); i++) {
				row = sheet.getRow(i);
				if (row == null) {
					continue;
				}

				List<Object> linked = new LinkedList<Object>();
				for (int j = row.getFirstCellNum(); j <= row.getLastCellNum(); j++) {
					cell = row.getCell(j);
					try{
						if (cell == null || StringUtils.isNullOrEmptyStr(cell.toString())) {
							linked.add("");
							continue;
						}
					}catch(Exception e){
						linked.add("");
						continue;
					}

					DecimalFormat df = new DecimalFormat("0");// 格式化 number String
					// 字符
					SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");// 格式化日期字符串
					DecimalFormat nf = new DecimalFormat("0");// 格式化数字
					switch (cell.getCellType()) {
						case STRING:
							// System.out.println(i+"行"+j+" 列 is String type");
							value = cell.getStringCellValue();
							break;
						case NUMERIC:
							// System.out.println(i+"行"+j+" 列 is Number type ;
							// DateFormt:"+cell.getCellStyle().getDataFormatString());
							if ("@".equals(cell.getCellStyle().getDataFormatString())) {
								value = df.format(cell.getNumericCellValue());
							} else if ("General".equals(cell.getCellStyle().getDataFormatString())) {
								value = nf.format(cell.getNumericCellValue());
							} else {
								value = sdf.format(HSSFDateUtil.getJavaDate(cell.getNumericCellValue()));
							}
							break;
						case BOOLEAN:
							// System.out.println(i+"行"+j+" 列 is Boolean type");
							value = cell.getBooleanCellValue();
							break;
						case BLANK:
							// System.out.println(i+"行"+j+" 列 is Blank type");
							value = "";
							break;
						default:
							// System.out.println(i+"行"+j+" 列 is default type");
							value = cell.toString();
					}
					if (value == null || "".equals(value)) {
						continue;
					}
					linked.add(value);
				}
				list.add(linked);
			}
		}finally {
			if(xwb!=null){
				xwb.close();
			}
			if(fis!=null){
				fis.close();
			}
		}
		return list;
	}
}
